#ifndef AIDETECTBINGXINGKERNEL_H
#define AIDETECTBINGXINGKERNEL_H

#include <cuda_runtime.h>
#include <opencv2/core/cuda.hpp>
#include <device_launch_parameters.h>
#include <cuda_runtime_api.h>
//#include "/usr/local/cuda-10.2/targets/aarch64-linux/include/cuda_runtime_api.h"
//#include "/usr/local/cuda-10.2/targets/aarch64-linux/include/cuda_runtime.h"
//#include "/usr/local/cuda-10.2/targets/aarch64-linux/include/device_launch_parameters.h"


// 预处理核函数声明
__global__ void preprocessKernel(uchar3* src, int srcWidth, int srcHeight, int srcStep,
                               float* dst, int dstWidth, int dstHeight);

// 后处理核函数声明
__global__ void postprocessKernel(const float* output, int numClasses,
                                uchar* mask, int width, int height);
int test_cuda(void);
#endif // AIDETECTBINGXINGKERNEL_H
